import os

# 算每天搞了多少图的，适用于很多文件夹嵌套
def count_images(directory):
    # 初始化图片计数器
    image_count = 0

    # 遍历指定目录
    for root, dirs, files in os.walk(directory):
        for file in files:
            # 获取文件的扩展名
            ext = os.path.splitext(file)[-1].lower()
            # 如果扩展名为图片格式，则计数器加一
            if ext in ['.jpg', '.jpeg', '.png', '.gif']:
                image_count += 1

    return image_count

# 测试函数
print(count_images('D:/new'))
